# coding=utf-8
import datetime
from typing import Dict, Optional, Any, Iterator
import requests
from loguru import logger

import models_provider.impl.base_chat_open_ai as basechatopenai
import json

class QwenVLChatModel(basechatopenai.MaxKBBaseModel, basechatopenai.BaseChatOpenAI):

    @staticmethod
    def stream(
            **kwargs
    ) -> Iterator[Any]:
        url = "https://your-qwen-vl-api.com/stream"  # Replace with actual API endpoint
        headers = {
            "Content-Type": "application/json",
            # Add other necessary headers here
        }
        
        now = datetime.datetime.now()
        api_request_time = f"{now.strftime('%Y-%m-%d %H:%M:%S')}"
        
        data = {
            **{
                "model": "<your-model>",
                "max_tokens": <max tokens>,
                "temperature": <temperature>,
                "top_p": <top p>
            },
            **{k: v for k, v in kwargs.items() if any(v)},
            "stream": True,
        }

        try:
            response = requests.post(url, headers=headers, json=data, stream=True)
            
            if response.status_code != 200:
                raise Exception(f"Failed to get response: {response.text}")

            for line in response.iter_lines(decode_unicode=True):
                if not line:
                    continue
                
                event = json.loads(line.split('data:', maxsplit=1)[1])
                
                delta = event.get("delta", {})
                content = delta.get("content", "")
                
                if content:
                    yield AIMessage(content=content)
                    
        except (requests.exceptions.RequestException, ValueError) as e:
            logger.error(f"An error occurred while fetching data: {e}")